package edu.hubu.talentmanagementsystem.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.google.common.base.CaseFormat;
import edu.hubu.talentmanagementsystem.mapper.CompanyStarMapper;
import edu.hubu.talentmanagementsystem.mapper.context.MapperContext;
import edu.hubu.talentmanagementsystem.model.ext.CompanyStar;
import edu.hubu.talentmanagementsystem.service.ICompanyStarService;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
import edu.hubu.talentmanagementsystem.common.EncodeOperation;
import edu.hubu.talentmanagementsystem.common.EncodeParam;

import java.util.List;
import java.util.Optional;

/**
 * @author moonlan
 * date 2021/3/2 下午8:45
 */
@SuppressWarnings(value = "unused")
@Service
@Transactional(rollbackFor = Exception.class)
public class CompanyStarServiceImpl extends ServiceImpl<CompanyStarMapper, CompanyStar> implements ICompanyStarService {

    private final MapperContext context;

    public CompanyStarServiceImpl(MapperContext context) {this.context = context;}

    @Override
    public boolean deleteById(Integer id) {
        return context.getCompanyStarMapper().deleteById(id) > 0;
    }

    @Override
    @EncodeOperation
    public Optional<CompanyStar> insert(@EncodeParam CompanyStar record) {
        int insert = context.getCompanyStarMapper().insert(record);
        if (insert > 0) {
            return Optional.of(context.getCompanyStarMapper().selectById(record.primaryKeyValue()));
        }
        return Optional.empty();
    }

    @Override
    public Optional<CompanyStar> selectById(Integer id) {
        QueryWrapper<CompanyStar> wrapper = new QueryWrapper<>();
        LambdaQueryWrapper<CompanyStar> lambdaQueryWrapper = getColumnWrapper(wrapper.lambda()).eq(CompanyStar::getCompanyStarId, id);
        return Optional.of(context.getCompanyStarMapper().selectOne(lambdaQueryWrapper));
    }

    @Override
    public List<CompanyStar> selectAll(String order) {
        QueryWrapper<CompanyStar> wrapper = new QueryWrapper<>();
        if (!StringUtils.hasText(order)) {
            LambdaQueryWrapper<CompanyStar> select = getColumnWrapper(wrapper.lambda());
            return context.getCompanyStarMapper().selectList(select);
        }
        wrapper.orderByAsc(CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, order));
        LambdaQueryWrapper<CompanyStar> select = getColumnWrapper(wrapper.lambda());
        return context.getCompanyStarMapper().selectList(select);
    }

    @Override
    public Page<CompanyStar> selectAllPaged( Integer currentPage,
             Integer pageSize, String order) {
        QueryWrapper<CompanyStar> wrapper = new QueryWrapper<>();
        Page<CompanyStar> page = new Page<>(currentPage, pageSize);
        if (!StringUtils.hasText(order)) {
            LambdaQueryWrapper<CompanyStar> select = getColumnWrapper(wrapper.lambda());
            return context.getCompanyStarMapper().selectPage(page, select);
        }
        wrapper.orderByAsc(CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, order));
        LambdaQueryWrapper<CompanyStar> select = getColumnWrapper(wrapper.lambda());
        return context.getCompanyStarMapper().selectPage(page, select);
    }

    private LambdaQueryWrapper<CompanyStar> getColumnWrapper(LambdaQueryWrapper<CompanyStar> wrapper) {
        return wrapper.select(
                CompanyStar::getCompanyStarId
                    ,
                CompanyStar::getCompanyStarCompanyId
                    ,
                CompanyStar::getCompanyStarStaffId
        );
    }

    @Override
    public Page<CompanyStar> selectLikePaged( Integer currentPage,
             Integer pageSize, CompanyStar record, String order) {
        Page<CompanyStar> page = new Page<>(currentPage, pageSize);
        QueryWrapper<CompanyStar> wrapper = new QueryWrapper<>();
        if (!StringUtils.hasText(order)) {
            return context.getCompanyStarMapper().selectPage(page, getColumnWrapper(getQueryWrapper(wrapper, record)));
        }
        wrapper.orderByAsc(CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, order));
        return context.getCompanyStarMapper().selectPage(page, getColumnWrapper(getQueryWrapper(wrapper, record)));
    }

    private LambdaQueryWrapper<CompanyStar> getQueryWrapper(QueryWrapper<CompanyStar>wrapper, CompanyStar record) {
        LambdaQueryWrapper<CompanyStar> lambda = wrapper.lambda();
        if (record.getCompanyStarId() != null) {
            lambda = lambda.eq(CompanyStar::getCompanyStarId,record.getCompanyStarId());
        }
        if (record.getCompanyStarCompanyId() != null) {
            lambda = lambda.eq(CompanyStar::getCompanyStarCompanyId,record.getCompanyStarCompanyId());
        }
        if (record.getCompanyStarStaffId() != null) {
            lambda = lambda.eq(CompanyStar::getCompanyStarStaffId,record.getCompanyStarStaffId());
        }
        return lambda;
    }

    public int count(CompanyStar record) {
        return context.getCompanyStarMapper().selectCount(getQueryWrapper(new QueryWrapper<>(), record));
    }

    @Override
    @EncodeOperation
    public Optional<CompanyStar> updateById(Integer id, @EncodeParam CompanyStar record) {
        record.setCompanyStarId(id);
        int update = context.getCompanyStarMapper().updateById(record);
        if (update > 0) {
            QueryWrapper<CompanyStar> wrapper = new QueryWrapper<>();
            LambdaQueryWrapper<CompanyStar> lambdaQueryWrapper = getColumnWrapper(wrapper.lambda()).eq(CompanyStar::getCompanyStarId, id);
            return Optional.of(context.getCompanyStarMapper().selectOne(getColumnWrapper(lambdaQueryWrapper)));
        }

        return Optional.empty();
    }

    @Override
    public boolean cancelFollow(Integer staffId, Integer companyId) {
        return context.getCompanyStarMapper()
                .delete(new QueryWrapper<CompanyStar>()
                        .eq("company_star_staff_id", staffId)
                        .eq("company_star_company_id", companyId)) > 0;
    }
}
